from ultralytics import YOLO

model = YOLO("/home/ubuntu/ModelCompress/Session3/yolov8/ultralytics/cfg/models/v8/yolov8.yaml")


dataset = "/home/ubuntu/Project/car.yaml"
device = [0]
batch = 16 * len(device)
epochs = 300
patience = 0
imgsz = 640
cache = True
workers = 32
project = "/home/ubuntu/ModelCompress/Session3_Weights"
name = "YOLOv8"
profile = False



model.train(
    data = dataset,
    epochs = epochs,
    device = device,
    batch = batch,
    patience = patience,
    imgsz = imgsz,
    cache = cache,
    workers = workers,
    project = project,
    name = name,
    profile = profile,
)